@echo off

REM 设置数据集目录和工作空间
set DATASET_DIR="D:\machinelearn\final\panns_transfer_to_gtzan\dataset_root"
set WORKSPACE="D:\machinelearn\final\panns_transfer_to_gtzan"

REM 运行特征提取脚本
python utils\features.py pack_audio_files_to_hdf5 --dataset_dir=%DATASET_DIR% --workspace=%WORKSPACE% --mini_data

REM 设置预训练检查点路径
set PRETRAINED_CHECKPOINT_PATH="D:\machinelearn\final\panns_transfer_to_gtzan\models\Cnn14_mAP=0.431.pth"

REM 训练模型 Transfer_Cnn14
set CUDA_VISIBLE_DEVICES=3
python pytorch\main.py train --dataset_dir=%DATASET_DIR% --workspace=%WORKSPACE% --holdout_fold=1 --model_type="Transfer_Cnn14" --pretrained_checkpoint_path=%PRETRAINED_CHECKPOINT_PATH% --loss_type=clip_nll --augmentation="mixup" --learning_rate=1e-4 --batch_size=8 --resume_iteration=0 --stop_iteration=600 --cuda
